package com.chenhy.mail;

import com.chenhy.server.pojo.Employee;
import com.chenhy.server.pojo.MailConstants;
import com.rabbitmq.client.Channel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.amqp.support.AmqpHeaders;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.mail.MailProperties;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
import org.springframework.stereotype.Component;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import java.io.IOException;
import java.util.Date;

@Component
public class MailReceiver {

    private static final Logger LOGGER = LoggerFactory.getLogger(MailReceiver.class);

    @Autowired
    private JavaMailSender javaMailSender;

    @Autowired
    private MailProperties mailProperties;

    @Autowired
    private TemplateEngine templateEngine;

    @Autowired
    private RedisTemplate redisTemplate;

    @RabbitListener(queues = MailConstants.MAIL_QUEUE_NAME)
    public void handler(Message message, Channel channel){
        // 拿到发送的数据
        Employee employee = (Employee) message.getPayload();
        // 拿到消息序号
        MessageHeaders headers = message.getHeaders();
        Long tar = (Long) headers.get(AmqpHeaders.DELIVERY_TAG);
        // 拿到uuid
        String msgId = (String) headers.get("spring_returned_message_correlation");
        // 拿到redis的forHash
        HashOperations hashOperations = redisTemplate.opsForHash();


        try {
            // 判断redis的hash中包不包含key为本次消息的UUID的值
            if(hashOperations.entries("mail_log").containsKey(msgId)){
                // 如果包含，代表消息已经被消费过
                LOGGER.error("消息已经被消费", msgId);
                // 手动确认消息，tar：消息序号，布尔值：是否确认多条
                channel.basicAck(tar, false);
                return;
            }

            MimeMessage mimeMessage = javaMailSender.createMimeMessage();
            MimeMessageHelper helper = new MimeMessageHelper(mimeMessage);
            // 发件人
            helper.setFrom(mailProperties.getUsername());
            // 收件人
            helper.setTo(employee.getEmail());
            // 主题
            helper.setSubject("入职欢迎邮件");
            // 发送日期
            helper.setSentDate(new Date());
            // Thymeleaf上下文
            Context context = new Context();
            context.setVariable("name", employee.getName());
            context.setVariable("posName", employee.getPosition().getName());
            context.setVariable("joblevelName", employee.getJoblevel().getName());
            context.setVariable("departmentName", employee.getDepartment().getName());
            // 使用Thymeleaf模板引擎处理页面
            String mail = templateEngine.process("mail", context);
            helper.setText(mail,true);
            // 发送邮件
            javaMailSender.send(mimeMessage);
            LOGGER.info("邮件发送成功");
            // 这条消息的UUID存入redis
            hashOperations.put("mail_log", msgId, "OK");
            // 手动确认消息
            channel.basicAck(tar, false);
        } catch (Exception e) {
            // 手动确认消息，tar：消息序号，布尔值false：是否确认多条，布尔值true：该消息是否退回队列
            try {
                channel.basicNack(tar, false, true);
            } catch (IOException ex) {
                ex.printStackTrace();
            }
            LOGGER.error("邮件发送失败",e.getMessage());
        }




    }

}
